import numpy as npimport matplotlib